package com.len.core;

import com.len.entity.SysRole;
import com.len.entity.SysRoleUser;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Utils {

    /**
     * 求两个集合交集，（map）
     *
     * @param arr1
     * @param arr2
     * @return
     */
    public static List<String> mapForList(List<SysRole> arr1, List<SysRoleUser> arr2) {

        List<String> resultList = new ArrayList<>();
        Map<String, String> map = new HashMap<>();
        arr1.forEach(a1 -> {
            map.put(a1.getId(), a1.getId());
        });

        arr2.forEach(a2 -> {
            String roleId = map.get(a2.getRoleId());
            if (roleId != null) {
                resultList.add(roleId);
            }
        });
        return resultList;
    }


}
